from app.project.doc_to_recommendation.llm.model.HttpChat_base_model import HttpChatBaseModel
from app.project.doc_to_recommendation.llm.register.llm_register import LLM_REGISTER

@LLM_REGISTER.register_model("http_chat_gen_model")
class HttpChatGenModel(HttpChatBaseModel):

    def __init__(self, config: dict):
        super().__init__(config)

    def bind_tools_df(self):
        pass

    def agent_calls(self, text, image=None, prompt=None):
        result = super().agent_calls(text, image, prompt)
        # content = result.get('txBody', {}).get('txEntity', {}).get('choices', [{}])[0].get('message', {}).get('content', '')
        return result
